/aosp_15_r20/external/federated-compute/fcp/aggregation/tensorflow/ |
H A D | converters_test.cc | 116 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 D | converters.h | 54 StatusOr<Tensor> ConvertTensor(std::unique_ptr<::tensorflow::Tensor> tensor);
|
H A D | checkpoint_reader.cc | 89 return ConvertTensor(std::move(tensor)); in GetTensor()
|
H A D | converters.cc | 129 StatusOr<Tensor> ConvertTensor(std::unique_ptr<tf::Tensor> tensor) { in ConvertTensor() function
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
H A D | convert_tensor_test.cc | 94 auto value_or_status = ConvertTensor(tensor, &b); in TEST() 117 auto value_or = ConvertTensor(tensor, &b); in VerifyConversion()
|
H A D | convert_tensor.h | 38 StatusOr<mlir::ElementsAttr> ConvertTensor(const Tensor& input_tensor,
|
H A D | convert_tensor.cc | 116 StatusOr<ElementsAttr> ConvertTensor(const Tensor& input_tensor, in ConvertTensor() function 233 return ConvertTensor(t, builder); in ConvertTensorProto()
|
H A D | eval_util.cc | 141 auto attr_or = ConvertTensor(tensor, &builder); in EvaluateOperation()
|
H A D | compile_mlir_util.cc | 740 ConvertTensor(xla_arg.constant_value, &builder)); in RewriteWithArgs()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/ |
H A D | tf_cpurt_kernels.cc | 283 converter.AddConversion(ReturnAsyncStridedMemref<ConvertTensor>); in ExecuteImpl() 284 converter.AddConversion(ReturnStridedMemref<ConvertTensor>); in ExecuteImpl()
|
H A D | tf_jitrt_test.cc | 44 ReturnStridedMemref<ConvertTensor>>;
|
H A D | tf_cpurt.h | 87 struct ConvertTensor { struct
|
H A D | tf_jitrt.h | 106 struct ConvertTensor { struct
|
H A D | tf_jitrt_kernels.cc | 581 ReturnStridedMemref<ConvertTensor>>;
|
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/ |
H A D | convert_tensor.h | 38 tensorflow::StatusOr<ElementsAttr> ConvertTensor(
|
H A D | convert_tensor.cc | 113 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 D | eval_utils.cc | 152 ConvertTensor(*(op_context.mutable_output(i)), builder); in EvaluateOperation()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
H A D | initialize_variables_in_session_init.cc | 42 tensorflow::ConvertTensor(*tensor, &builder); in InitializeVariable()
|
H A D | lift_variables.cc | 132 StatusOr<ElementsAttr> tensor_attr_or = ConvertTensor(tensor, &builder); in LiftVariablesFromSession()
|
H A D | tensor_array_ops_decomposition.cc | 221 tensorflow::ConvertTensor(scalar_tensor, &builder).ValueOrDie()); in HandleTensorArrayV3Op() 317 tensorflow::ConvertTensor(lengths_tensor, &builder).ValueOrDie())); in HandleTensorArrayConcatV3Op()
|
H A D | tf_saved_model_freeze_variables.cc | 53 tensorflow::ConvertTensor(tensor, &builder); in GetTensorValueAsElementsAttr()
|
H A D | tpu_variable_runtime_reformatting.cc | 444 tensorflow::ConvertTensor(default_key_tensor, &builder).ValueOrDie()); in HandleReplicateOp()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/gpu/ |
H A D | conversion_function.cc | 163 auto host_tensor_ref = tfrt::ConvertTensor( in ConvertRuntimeFallbackTensorToDenseGpuTensor()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
H A D | lower_static_tensor_list.cc | 289 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 D | runtime_fallback_kernels.cc | 855 return tfrt::ConvertTensor(exec_ctx, tensor, in CoreRTTensorHandleToFallbackTensorInternal()
|