Home
last modified time | relevance | path

Searched refs:tensor_shape_value (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/
H A Dnested_structure_coder.py358 def do_encode(self, tensor_shape_value, encode_fn): argument
361 encoded_tensor_shape.tensor_shape_value.CopyFrom(
362 tensor_shape_value.as_proto())
370 return tensor_shape.TensorShape(value.tensor_shape_value)
405 shape=encode_fn(tensor_spec_value.shape).tensor_shape_value,
418 tensor_shape_value=value.tensor_spec_value.shape)),
436 shape=encode_fn(bounded_tensor_spec_value.shape).tensor_shape_value,
453 struct_pb2.StructuredValue(tensor_shape_value=btsv.shape)),
H A Dnested_structure_coder_test.py86 expected_tensor_shape = expected_list.values.add().tensor_shape_value
/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/
Dstruct.pb.cc295 static const ::tensorflow::TensorShapeProto& tensor_shape_value(const StructuredValue* msg);
310 StructuredValue::_Internal::tensor_shape_value(const StructuredValue* msg) { in tensor_shape_value() function in tensorflow::StructuredValue::_Internal
356 …cturedValue::set_allocated_tensor_shape_value(::tensorflow::TensorShapeProto* tensor_shape_value) { in set_allocated_tensor_shape_value() argument
359 if (tensor_shape_value) { in set_allocated_tensor_shape_value()
362 reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(tensor_shape_value)); in set_allocated_tensor_shape_value()
364 tensor_shape_value = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage( in set_allocated_tensor_shape_value()
365 message_arena, tensor_shape_value, submessage_arena); in set_allocated_tensor_shape_value()
368 _impl_.kind_.tensor_shape_value_ = tensor_shape_value; in set_allocated_tensor_shape_value()
903 InternalWriteMessage(31, _Internal::tensor_shape_value(this), in _InternalSerialize()
904 _Internal::tensor_shape_value(this).GetCachedSize(), target, stream); in _InternalSerialize()
Dstruct.pb.h357 const ::tensorflow::TensorShapeProto& tensor_shape_value() const; in Swap()
360 void set_allocated_tensor_shape_value(::tensorflow::TensorShapeProto* tensor_shape_value); in Swap()
366 ::tensorflow::TensorShapeProto* tensor_shape_value); in Swap()
2370 inline const ::tensorflow::TensorShapeProto& StructuredValue::tensor_shape_value() const { in tensor_shape_value() function
2385 …unsafe_arena_set_allocated_tensor_shape_value(::tensorflow::TensorShapeProto* tensor_shape_value) { in unsafe_arena_set_allocated_tensor_shape_value() argument
2387 if (tensor_shape_value) { in unsafe_arena_set_allocated_tensor_shape_value()
2389 _impl_.kind_.tensor_shape_value_ = tensor_shape_value; in unsafe_arena_set_allocated_tensor_shape_value()
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Dstruct.proto57 tensorflow.TensorShapeProto tensor_shape_value = 31; field