Searched refs:size_shape (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | resize_bilinear_tester.cc | 137 const std::array<int32_t, 1> size_shape{ in CreateTfLiteModel() local
147 builder.CreateVector<int32_t>(size_shape.data(), size_shape.size()), in CreateTfLiteModel()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | slice_test.cc | 42 std::initializer_list<int> size_shape, in SliceOpModel() argument
53 size_ = AddConstInput(GetTensorType<index_type>(), size_data, size_shape); in SliceOpModel()
58 BuildInterpreter({input_shape, begin_shape, size_shape}); in SliceOpModel()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.cc | 131 ArrayRef<int64_t> size_shape, in SliceRankedTensor() argument
141 return CreateF32SplatConst(builder, size_shape, 0, location); in SliceRankedTensor()
151 CreateI64DenseConst(builder, size_shape, size_values, location); in SliceRankedTensor()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | list_ops.cc | 454 shape_inference::ShapeHandle size_shape = c->input(1); in __anonabedecb60e02() local
456 TF_RETURN_IF_ERROR(c->WithRank(size_shape, 0, &unused)); in __anonabedecb60e02()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | shape_op.cc | 120 const TensorShape size_shape = ctx->InputShape("size"); in Compile() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 823 const Shape& size_shape = size_array.shape(); in ProcessTensorFlowSplitVOperator() local
828 CHECK_EQ(size_shape.dimensions_count(), 1) << "size_splits must be 1-D"; in ProcessTensorFlowSplitVOperator()
|