Searched refs:ConvertTensor (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_tensor_test.cc | 89 auto value_or_status = ConvertTensor(tensor, &b); in TEST()
112 auto value_or = ConvertTensor(tensor, &b); in VerifyConversion()
|
| D | convert_tensor.cc | 121 StatusOr<ElementsAttr> ConvertTensor(const Tensor& input_tensor, in ConvertTensor() function
176 return ConvertTensor(t, builder); in ConvertTensorProto()
391 return ConvertTensor(tensor, &builder); in DecodeOpaqueTensor()
|
| D | convert_tensor.h | 38 StatusOr<mlir::ElementsAttr> ConvertTensor(const Tensor& input_tensor,
|
| D | eval_util.cc | 141 auto attr_or = ConvertTensor(tensor, &builder); in EvaluateOperation()
|
| D | compile_mlir_util.cc | 563 ConvertTensor(xla_arg.constant_value, &builder)); in RewriteWithArgs()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lift_variables.cc | 132 StatusOr<ElementsAttr> tensor_attr_or = ConvertTensor(tensor, &builder); in LiftVariablesFromSession()
|
| D | tpu_variable_runtime_reformatting.cc | 491 tensorflow::ConvertTensor(default_key_tensor, &builder).ValueOrDie()); in HandleReplicateOp()
|
| D | tensor_array_ops_decomposition.cc | 222 tensorflow::ConvertTensor(scalar_tensor, &builder).ValueOrDie()); in HandleTensorArrayV3Op()
318 tensorflow::ConvertTensor(lengths_tensor, &builder).ValueOrDie())); in HandleTensorArrayConcatV3Op()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | lower_static_tensor_list.cc | 221 auto attr_or = tensorflow::ConvertTensor(tensor, &rewriter); in matchAndRewrite()
233 auto attr_or = tensorflow::ConvertTensor(tensor, &rewriter); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 3219 TF_ASSIGN_OR_RETURN(auto value_attr, ConvertTensor(value, &builder)); in CreateSavedModelIR()
|