Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Daccumulate_n_optimizer.cc156 PartialTensorShape partial_shape(shape); in RewriteIntoTempVariable() local
/external/tensorflow/tensorflow/c/eager/
Dc_api_unified_experimental.cc148 tensorflow::PartialTensorShape partial_shape; in TF_AddFunctionParameter() local
/external/tensorflow/tensorflow/core/framework/
Dshape_inference.cc863 const PartialTensorShape& partial_shape, ShapeHandle* out) { in MakeShapeFromPartialTensorShape()
888 PartialTensorShape partial_shape(proto); in MakeShapeFromShapeProto() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtensor_list_ops.cc122 PartialTensorShape partial_shape; in TryGetElementShapeFromInput() local
/external/tensorflow/tensorflow/core/kernels/
Dfunctional_ops.cc871 PartialTensorShape partial_shape; in FakeParamOp() local
/external/tensorflow/tensorflow/c/
Dc_api_test.cc2190 const int64_t partial_shape[] = {17, -1}; in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.cc874 const Shape& partial_shape = rhs->operand(1)->shape(); in HandleAdd() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc6473 PartialTensorShape partial_shape; in ConvertSegmentToGraphDef() local
Dconvert_nodes_test.cc1831 std::vector<int32> partial_shape; local