| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer.cc | 158 graph_input_node_info.add_shape(dim); in LoadGraphFromProto()
183 graph_output_node_info.add_shape(dim); in LoadGraphFromProto()
489 const_node_info.add_shape(shape_array[0]); in RegisterConstantNode()
490 const_node_info.add_shape(shape_array[1]); in RegisterConstantNode()
491 const_node_info.add_shape(shape_array[2]); in RegisterConstantNode()
492 const_node_info.add_shape(shape_array[3]); in RegisterConstantNode()
520 const_node_info.add_shape(static_cast<int64>(shape[0])); in RegisterConstantShape()
521 const_node_info.add_shape(static_cast<int64>(shape[1])); in RegisterConstantShape()
522 const_node_info.add_shape(static_cast<int64>(shape[2])); in RegisterConstantShape()
523 const_node_info.add_shape(static_cast<int64>(shape[3])); in RegisterConstantShape()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | hoist_random_uniform.cc | 83 (*random_dataset.mutable_attr())["output_shapes"].mutable_list()->add_shape(); in MakeRandomDataset()
103 ->add_shape() in MakeBatchTwo()
|
| D | map_vectorization.cc | 244 TensorShapeProto* shape = output_shapes_attr.mutable_list()->add_shape(); in AddNewBatchNode()
|
| D | auto_shard.cc | 173 ->add_shape(); in AddShardNode()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | attr_value_util.cc | 528 v.AsProto(out->mutable_list()->add_shape()); in SetAttrValue()
535 *out->mutable_list()->add_shape() = v; in SetAttrValue()
543 v.AsProto(out->mutable_list()->add_shape()); in SetAttrValue()
|
| D | graph_to_functiondef.cc | 444 *(value.mutable_list()->add_shape()) = attr.second.shape(); in GraphToFunctionDef()
451 *(value.mutable_list()->add_shape()) = in GraphToFunctionDef()
|
| D | function_test.cc | 1508 TensorShapeProto* shape_proto = shape_attr.mutable_list()->add_shape(); in TEST()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | export_utils.cc | 202 *list->add_shape() = nested_value.shape(); in ConvertAttribute()
228 *list->add_shape() = attr_val.shape(); in ConvertAttribute()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | export_tf_dialect_op.cc | 75 TensorShapeProto& tshape = *shape_list.add_shape(); in SetShapeAttribute()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 506 [response] { return response->add_shape(); })); in RunComponentFunction()
545 [queue_response] { return queue_response->add_shape(); }, in ExecuteOp()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_constants_test.cc | 146 auto* shape_proto = shape_attr.mutable_list()->add_shape(); in TestPreserveOutputShapes()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | import_tensorflow_test.cc | 90 tensorflow::TensorShapeProto* shape = shapes->add_shape(); in BuildNode()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_test.cc | 684 auto* shape = output_shapes.mutable_list()->add_shape(); in TEST_F()
|
| D | layout_optimizer.cc | 884 auto output_shape = attr_output_shape.mutable_list()->add_shape(); in AddNodeTranspose()
|
| D | generic_layout_optimizer_transposer.cc | 353 *attr_output_shape.mutable_list()->add_shape() = fanin_shape; in CreateTransposeNode()
|
| D | constant_folding_test.cc | 4174 output_shapes.mutable_list()->add_shape(); in TEST_F()
4176 TensorShapeProto* g_shape = output_shapes.mutable_list()->add_shape(); in TEST_F()
|
| D | constant_folding.cc | 2499 ->add_shape(); in SimplifyCase()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 199 TF_ASSERT_OK_AND_ASSIGN(auto add_shape, b.GetShape(add)); in TEST_F()
200 EXPECT_TRUE(ShapeUtil::Equal(add_shape, x_shape)); in TEST_F()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 820 output_attr.mutable_list()->add_shape()->Swap(&proto); in UpdateFunction()
836 shape_handle, shape_attr.mutable_list()->add_shape()); in UpdateFunction()
2735 TensorShapeProto* proto = attr_output_shape.mutable_list()->add_shape(); in AnnotateOutputShapes()
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_node.h | 386 node_.add_shape(s); in ToProto()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | serialization.cc | 474 tensor_builder.add_shape(shape_fb); in Encode()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | serialization_base_generated.h | 1443 void add_shape(flatbuffers::Offset<tflite::gpu::data::BHWDC> shape) { in add_shape() function
1470 builder_.add_shape(shape);
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 386 *shape_attr_value.mutable_list()->add_shape() = shape_proto; in SetArgShape()
|
| /external/tensorflow/tensorflow/lite/schema/ |
| D | schema_generated.h | 3771 void add_shape(flatbuffers::Offset<flatbuffers::Vector<int32_t>> shape) {
3823 builder_.add_shape(shape);
|