Home
last modified time | relevance | path

Searched refs:dynamic_shapes (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_util.cc79 Status DynamicShapesToTensorShapes(const OpInputList& dynamic_shapes, in DynamicShapesToTensorShapes() argument
81 shapes->resize(dynamic_shapes.size()); in DynamicShapesToTensorShapes()
82 for (int i = 0; i < dynamic_shapes.size(); ++i) { in DynamicShapesToTensorShapes()
84 ShapeTensorToTensorShape(dynamic_shapes[i], &(*shapes)[i])); in DynamicShapesToTensorShapes()
89 Status DynamicShapesToTensorShapes(const InputList& dynamic_shapes, in DynamicShapesToTensorShapes() argument
91 shapes->resize(dynamic_shapes.end() - dynamic_shapes.begin()); in DynamicShapesToTensorShapes()
93 for (auto& dynamic_shape : dynamic_shapes) { in DynamicShapesToTensorShapes()
Dtpu_op_util.cc29 const std::vector<tensorflow::TensorShape>& dynamic_shapes) { in CreateShapePrefix() argument
31 for (const TensorShape& shape : dynamic_shapes) { in CreateShapePrefix()
107 const std::vector<TensorShape>& dynamic_shapes, in CreateCompilationCacheKey() argument
111 std::string shapes_prefix = CreateShapePrefix(dynamic_shapes); in CreateCompilationCacheKey()
Dtpu_util.h55 Status DynamicShapesToTensorShapes(const OpInputList& dynamic_shapes,
57 Status DynamicShapesToTensorShapes(const InputList& dynamic_shapes,
Dtpu_compile_op_common.cc407 OpInputList dynamic_shapes; in GetDynamicShapes() local
408 TF_RETURN_IF_ERROR(ctx->input_list("dynamic_shapes", &dynamic_shapes)); in GetDynamicShapes()
410 shapes->resize(dynamic_shapes.size()); in GetDynamicShapes()
411 for (int i = 0; i < dynamic_shapes.size(); ++i) { in GetDynamicShapes()
413 tpu::ShapeTensorToTensorShape(dynamic_shapes[i], &(*shapes)[i])); in GetDynamicShapes()
594 const std::vector<TensorShape>& dynamic_shapes, in CompileLocallyAndFillHostCache() argument
600 ComputeArgumentShapes(metadata_, dynamic_shapes, &arg_shapes)); in CompileLocallyAndFillHostCache()
641 std::vector<TensorShape> dynamic_shapes; in ComputeInternal() local
642 TF_RETURN_IF_ERROR(GetDynamicShapes(ctx, &dynamic_shapes)); in ComputeInternal()
654 mlir_module_fingerprint_, guaranteed_constants, dynamic_shapes, metadata_, in ComputeInternal()
[all …]
Dtpu_op_util.h38 const std::vector<TensorShape>& dynamic_shapes,
Dtpu_compile_op_common.h122 const std::vector<TensorShape>& dynamic_shapes,
133 const std::vector<TensorShape>& dynamic_shapes, in LookupPersistentCompilationCacheAndFillCaches() argument
Dtpu_compile_op_support.cc442 const std::vector<TensorShape>& dynamic_shapes, in ComputeArgumentShapes() argument
461 TF_RET_CHECK(dynamic_shape_pos < dynamic_shapes.size()) in ComputeArgumentShapes()
463 shape = dynamic_shapes[dynamic_shape_pos++]; in ComputeArgumentShapes()
474 TF_RET_CHECK(dynamic_shape_pos == dynamic_shapes.size()) in ComputeArgumentShapes()
Dtpu_compile_op_support.h157 const std::vector<TensorShape>& dynamic_shapes,
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DTPUCompile.pbtxt4 name: "dynamic_shapes"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_TPUCompile.pbtxt9 'dynamic_shapes' contains dynamic shapes of arguments whose shapes were not
/external/tensorflow/tensorflow/lite/java/
DBUILD273 "//tensorflow/lite:testdata/dynamic_shapes.bin",
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td18522 'dynamic_shapes' contains dynamic shapes of arguments whose shapes were not
18532 Variadic<TF_Int64Tensor>:$dynamic_shapes,
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt53720 name: "dynamic_shapes"