Home
last modified time | relevance | path

Searched defs:partial_shape (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device_util.cc118 PartialTensorShape partial_shape = r.dtypes_and_shapes().begin()->shape; in BroadcastResourceTensor() local
670 tensorflow::PartialTensorShape partial_shape; in PrepareGraphForMlir() local
696 tensorflow::PartialTensorShape partial_shape; in PrepareGraphForMlir() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Daccumulate_n_optimizer.cc156 PartialTensorShape partial_shape(shape); in RewriteIntoTempVariable() local
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dc_api_unified_experimental.cc148 tensorflow::PartialTensorShape partial_shape; in TF_AddFunctionParameter() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dshape_inference.cc883 const PartialTensorShape& partial_shape, ShapeHandle* out) { in MakeShapeFromPartialTensorShape()
921 PartialTensorShape partial_shape(proto); in MakeShapeFromShapeProto() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dtensor_list_ops.cc124 PartialTensorShape partial_shape; in TryGetElementShapeFromInput() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_functional_ops.cc440 auto partial_shape = PartialTensorShape::MakePartialShape( in GetInputOutputInfo() local
2052 auto partial_shape = PartialTensorShape::MakePartialShape( in InferShapesWithResourceVar() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfunctional_ops.cc871 PartialTensorShape partial_shape; in FakeParamOp() local
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dc_api_test.cc2221 const int64_t partial_shape[] = {17, -1}; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes_test.cc1835 std::vector<int32> partial_shape; local
3129 std::vector<int> partial_shape(p.shape_a.size(), -1); in TestMatMulHelper() local
H A Dconvert_nodes.cc6039 PartialTensorShape partial_shape; in ConvertSegmentToGraphDef() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dalgebraic_simplifier.cc756 const Shape& partial_shape = rhs->operand(1)->shape(); in HandleAdd() local