Searched refs:ConvertTensor (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_tensor.h | 38 StatusOr<mlir::ElementsAttr> ConvertTensor(const Tensor& input_tensor,
|
| D | convert_tensor_test.cc | 94 auto value_or_status = ConvertTensor(tensor, &b); in TEST()
117 auto value_or = ConvertTensor(tensor, &b); in VerifyConversion()
|
| D | convert_tensor.cc | 120 StatusOr<ElementsAttr> ConvertTensor(const Tensor& input_tensor, in ConvertTensor() function
240 return ConvertTensor(t, builder); in ConvertTensorProto()
507 return ConvertTensor(tensor, &builder); in DecodeOpaqueTensor()
|
| D | eval_util.cc | 141 auto attr_or = ConvertTensor(tensor, &builder); in EvaluateOperation()
|
| D | compile_mlir_util.cc | 591 ConvertTensor(xla_arg.constant_value, &builder)); in RewriteWithArgs()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/ |
| D | tf_cpurt.h | 87 struct ConvertTensor { struct
|
| D | tf_cpurt_kernels.cc | 283 converter.AddConversion(ReturnAsyncStridedMemref<ConvertTensor>); in ExecuteImpl()
284 converter.AddConversion(ReturnStridedMemref<ConvertTensor>); in ExecuteImpl()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lift_variables.cc | 131 StatusOr<ElementsAttr> tensor_attr_or = ConvertTensor(tensor, &builder); in LiftVariablesFromSession()
|
| D | initialize_variables_in_session_init.cc | 66 tensorflow::ConvertTensor(*tensor, &builder); in InitializeVariable()
|
| D | tf_saved_model_freeze_variables.cc | 51 tensorflow::ConvertTensor(tensor, &builder); in GetTensorValueAsElementsAttr()
|
| D | tensor_array_ops_decomposition.cc | 221 tensorflow::ConvertTensor(scalar_tensor, &builder).ValueOrDie()); in HandleTensorArrayV3Op()
317 tensorflow::ConvertTensor(lengths_tensor, &builder).ValueOrDie())); in HandleTensorArrayConcatV3Op()
|
| D | tpu_variable_runtime_reformatting.cc | 500 tensorflow::ConvertTensor(default_key_tensor, &builder).ValueOrDie()); in HandleReplicateOp()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/gpu/ |
| D | conversion_function.cc | 163 auto host_tensor_ref = tfrt::ConvertTensor( in ConvertRuntimeFallbackTensorToDenseGpuTensor()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | lower_static_tensor_list.cc | 290 auto attr_or = tensorflow::ConvertTensor(tensor, &rewriter); in matchAndRewrite()
302 auto attr_or = tensorflow::ConvertTensor(tensor, &rewriter); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | runtime_fallback_kernels.cc | 854 return tfrt::ConvertTensor(exec_ctx, tensor, in CoreRTTensorHandleToFallbackTensorInternal()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 3364 TF_ASSIGN_OR_RETURN(auto value_attr, ConvertTensor(value, &builder)); in CreateSavedModelIR()
|