Home
last modified time | relevance | path

Searched refs:ConvertTensor (Results 1 – 25 of 26) sorted by relevance

12

/aosp_15_r20/external/federated-compute/fcp/aggregation/tensorflow/
H A Dconverters_test.cc116 EXPECT_THAT(*ConvertTensor(std::move(tensor)), in TEST()
130 EXPECT_THAT(*ConvertTensor(std::move(tensor)), in TEST()
142 EXPECT_THAT(*ConvertTensor(std::move(tensor)), in TEST()
148 EXPECT_THAT(ConvertTensor(std::move(tensor)), IsCode(INVALID_ARGUMENT)); in TEST()
H A Dconverters.h54 StatusOr<Tensor> ConvertTensor(std::unique_ptr<::tensorflow::Tensor> tensor);
H A Dcheckpoint_reader.cc89 return ConvertTensor(std::move(tensor)); in GetTensor()
H A Dconverters.cc129 StatusOr<Tensor> ConvertTensor(std::unique_ptr<tf::Tensor> tensor) { in ConvertTensor() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dconvert_tensor_test.cc94 auto value_or_status = ConvertTensor(tensor, &b); in TEST()
117 auto value_or = ConvertTensor(tensor, &b); in VerifyConversion()
H A Dconvert_tensor.h38 StatusOr<mlir::ElementsAttr> ConvertTensor(const Tensor& input_tensor,
H A Dconvert_tensor.cc116 StatusOr<ElementsAttr> ConvertTensor(const Tensor& input_tensor, in ConvertTensor() function
233 return ConvertTensor(t, builder); in ConvertTensorProto()
H A Deval_util.cc141 auto attr_or = ConvertTensor(tensor, &builder); in EvaluateOperation()
H A Dcompile_mlir_util.cc740 ConvertTensor(xla_arg.constant_value, &builder)); in RewriteWithArgs()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/
H A Dtf_cpurt_kernels.cc283 converter.AddConversion(ReturnAsyncStridedMemref<ConvertTensor>); in ExecuteImpl()
284 converter.AddConversion(ReturnStridedMemref<ConvertTensor>); in ExecuteImpl()
H A Dtf_jitrt_test.cc44 ReturnStridedMemref<ConvertTensor>>;
H A Dtf_cpurt.h87 struct ConvertTensor { struct
H A Dtf_jitrt.h106 struct ConvertTensor { struct
H A Dtf_jitrt_kernels.cc581 ReturnStridedMemref<ConvertTensor>>;
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/
H A Dconvert_tensor.h38 tensorflow::StatusOr<ElementsAttr> ConvertTensor(
H A Dconvert_tensor.cc113 tensorflow::StatusOr<ElementsAttr> ConvertTensor(const Tensor& input_tensor, in ConvertTensor() function
235 return ConvertTensor(t, builder); in ConvertTensorProto()
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/utils/
H A Deval_utils.cc152 ConvertTensor(*(op_context.mutable_output(i)), builder); in EvaluateOperation()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dinitialize_variables_in_session_init.cc42 tensorflow::ConvertTensor(*tensor, &builder); in InitializeVariable()
H A Dlift_variables.cc132 StatusOr<ElementsAttr> tensor_attr_or = ConvertTensor(tensor, &builder); in LiftVariablesFromSession()
H A Dtensor_array_ops_decomposition.cc221 tensorflow::ConvertTensor(scalar_tensor, &builder).ValueOrDie()); in HandleTensorArrayV3Op()
317 tensorflow::ConvertTensor(lengths_tensor, &builder).ValueOrDie())); in HandleTensorArrayConcatV3Op()
H A Dtf_saved_model_freeze_variables.cc53 tensorflow::ConvertTensor(tensor, &builder); in GetTensorValueAsElementsAttr()
H A Dtpu_variable_runtime_reformatting.cc444 tensorflow::ConvertTensor(default_key_tensor, &builder).ValueOrDie()); in HandleReplicateOp()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/gpu/
H A Dconversion_function.cc163 auto host_tensor_ref = tfrt::ConvertTensor( in ConvertRuntimeFallbackTensorToDenseGpuTensor()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Dlower_static_tensor_list.cc289 auto attr_or = tensorflow::ConvertTensor(tensor, &rewriter); in matchAndRewrite()
301 auto attr_or = tensorflow::ConvertTensor(tensor, &rewriter); in matchAndRewrite()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_kernels.cc855 return tfrt::ConvertTensor(exec_ctx, tensor, in CoreRTTensorHandleToFallbackTensorInternal()

12