| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental.cc | 545 void TF_ShapeAndTypeListSetShape(TF_ShapeAndTypeList* shape_list, int index, in TF_ShapeAndTypeListSetShape() argument
547 DCHECK(index >= 0 && index < shape_list->num_items); in TF_ShapeAndTypeListSetShape()
548 TF_ShapeAndType& shape = shape_list->items[index]; in TF_ShapeAndTypeListSetShape()
556 void TF_ShapeAndTypeListSetUnknownShape(TF_ShapeAndTypeList* shape_list, in TF_ShapeAndTypeListSetUnknownShape() argument
558 DCHECK(index >= 0 && index < shape_list->num_items); in TF_ShapeAndTypeListSetUnknownShape()
559 TF_ShapeAndType& shape = shape_list->items[index]; in TF_ShapeAndTypeListSetUnknownShape()
565 void TF_ShapeAndTypeListSetDtype(TF_ShapeAndTypeList* shape_list, int index, in TF_ShapeAndTypeListSetDtype() argument
567 DCHECK(index >= 0 && index < shape_list->num_items); in TF_ShapeAndTypeListSetDtype()
568 TF_ShapeAndType& shape_and_type = shape_list->items[index]; in TF_ShapeAndTypeListSetDtype()
572 void TF_DeleteShapeAndTypeList(TF_ShapeAndTypeList* shape_list) { in TF_DeleteShapeAndTypeList() argument
[all …]
|
| D | c_api_experimental.h | 272 TF_ShapeAndTypeList* shape_list, int index, const int64_t* dims,
275 TF_ShapeAndTypeList* shape_list, int index);
277 TF_ShapeAndTypeList* shape_list, int index, TF_DataType dtype);
279 TF_ShapeAndTypeList* shape_list);
|
| /external/tensorflow/tensorflow/python/training/ |
| D | input.py | 670 def _merge_shapes(shape_list, enqueue_many): argument
671 shape_list = [tensor_shape.as_shape(s) for s in shape_list]
674 shape_list = [s.with_rank_at_least(1)[1:] for s in shape_list]
675 merged_shape = shape_list[0]
676 for s in shape_list[1:]:
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | lite.py | 636 shape_list = tensor.shape.as_list()
637 if None in shape_list[1:]:
641 _get_tensor_name(tensor), shape_list))
642 elif shape_list and shape_list[0] is None:
1279 shape_list = shape.as_list()
1280 if None in shape_list[1:]:
1284 _get_tensor_name(tensor), shape_list))
1285 elif shape_list and shape_list[0] is None:
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | tf_utils.py | 484 shape_list = shape.as_list()
485 shape_list[0] = None
486 dynamic_batch_spec._shape = tensor_shape.TensorShape(shape_list)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | export_tf_dialect_op.cc | 73 auto& shape_list = *value.mutable_list(); in SetShapeAttribute() local
75 TensorShapeProto& tshape = *shape_list.add_shape(); in SetShapeAttribute()
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_feed.py | 56 shape_list = np.array(tensor.shape.as_list())
57 quotients, remainders = np.divmod(shape_list, dims)
59 zip(quotients, remainders, dims, shape_list)):
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel_test.cc | 779 status.emplace_back("shape_list", context->GetAttr(attr_name, &shape_list)); in GetAttrKernel()
813 std::vector<TensorShape> shape_list; member in tensorflow::__anon843475bc0111::GetAttrKernel
920 ASSERT_EQ(2, get_attr_kernel->shape_list.size()); in TEST_F()
921 EXPECT_EQ("[2]", get_attr_kernel->shape_list[0].DebugString()); in TEST_F()
922 EXPECT_EQ("[4]", get_attr_kernel->shape_list[1].DebugString()); in TEST_F()
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | util.py | 432 shape_list = None if shape is None else shape_object.as_list()
435 shape_list,
440 shape_list,
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_strategy_test.py | 402 def shape_list(tensor): function
422 tensor_shape = shape_list(where)
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | backprop_test.py | 1121 shape_list = [[], None, [1, 2, 3], [None, None], [1, None, 3]]
1123 [tensor_shape.TensorShape(s).as_proto() for s in shape_list],
1124 backprop.make_attr([int(pywrap_tfe.TF_ATTR_SHAPE)], shape_list))
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding_test.cc | 4216 const auto& shape_list = node.attr().at("_output_shapes").list(); in TEST_F() local
4217 ASSERT_EQ(shape_list.shape_size(), 1); in TEST_F()
4218 EXPECT_EQ(shape_list.shape(0).dim_size(), 0); in TEST_F()
4221 EXPECT_EQ(shape_list.shape(0).unknown_rank(), false); in TEST_F()
4224 EXPECT_EQ(shape_list.shape(0).unknown_rank(), true); in TEST_F()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_ops_impl.py | 2476 shape_list = ([array_ops.ones(rank_1, dtype=dtypes.int32)] +
2478 multiples = array_ops.concat(shape_list, 0)
|