Home
last modified time | relevance | path

Searched defs:add_shape (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dremapper_test.cc1023 auto add_shape = ops::Placeholder::Shape({6}); in TEST_F() local
1088 auto add_shape = ops::Placeholder::Shape({1, 1, 1, 1, 6}); in TEST_F() local
1154 auto add_shape = ops::Placeholder::Shape({1, 1, 6}); in TEST_F() local
1915 auto add_shape = ops::Placeholder::Shape({8, 4, 32, 32, 128}); in TEST_F() local
1987 auto add_shape = ops::Placeholder::Shape({8, 4, 32, 32, 128}); in TEST_F() local
2059 auto add_shape = ops::Placeholder::Shape({8, 4, 32, 32, 128}); in TEST_F() local
H A Dmkl_remapper_test.cc759 auto add_shape = TensorShape({b0, 1, m, n}); in VerifyFused() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_batch_matmul_op.cc259 auto add_shape = ctx->input(3).shape(); in ExtendMklMatMulParams() local
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/framework/
Dgraph_transfer_info.pb.h1944 inline void GraphTransferConstNodeInfo::add_shape(::int64_t value) { in add_shape() function
2240 inline void GraphTransferGraphInputNodeInfo::add_shape(::int64_t value) { in add_shape() function
2356 inline void GraphTransferGraphOutputNodeInfo::add_shape(::int64_t value) { in add_shape() function
Dattr_value.pb.h1198 inline ::tensorflow::TensorShapeProto* AttrValue_ListValue::add_shape() { in add_shape() function
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/
Deager_service.pb.h5394 inline ::tensorflow::TensorShapeProto* QueueResponse::add_shape() { in add_shape() function
6430 inline ::tensorflow::TensorShapeProto* RunComponentFunctionResponse::add_shape() { in add_shape() function
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
H A Dserialization_base_generated.h1211 void add_shape(flatbuffers::Offset<tflite::gpu::data::BHWDC> shape) { in add_shape() function