Home
last modified time | relevance | path

Searched refs:shape_list (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/contrib/learn/python/learn/utils/
Dinput_fn_utils.py118 shape_list = t.get_shape().as_list()
119 shape_list[0] = default_batch_size
120 shape = tensor_shape.TensorShape(shape_list)
/external/tensorflow/tensorflow/lite/python/
Dlite.py261 shape_list = tensor.shape.as_list()
262 if None in shape_list[1:]:
265 "invalid shape '{1}'.".format(_tensor_name(tensor), shape_list))
266 elif shape_list and shape_list[0] is None:
686 shape_list = shape.as_list()
687 if None in shape_list[1:]:
690 "invalid shape '{1}'.".format(_tensor_name(tensor), shape_list))
691 elif shape_list and shape_list[0] is None:
/external/tensorflow/tensorflow/python/training/
Dinput.py671 def _merge_shapes(shape_list, enqueue_many): argument
672 shape_list = [tensor_shape.as_shape(s) for s in shape_list]
675 shape_list = [s.with_rank_at_least(1)[1:] for s in shape_list]
676 merged_shape = shape_list[0]
677 for s in shape_list[1:]:
/external/tensorflow/tensorflow/python/tpu/
Dtpu_feed.py56 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/
Dop_kernel_test.cc706 status.emplace_back("shape_list", context->GetAttr(attr_name, &shape_list)); in GetAttrKernel()
740 std::vector<TensorShape> shape_list; member in tensorflow::__anon74460fe50111::GetAttrKernel
847 ASSERT_EQ(2, get_attr_kernel->shape_list.size()); in TEST_F()
848 EXPECT_EQ("[2]", get_attr_kernel->shape_list[0].DebugString()); in TEST_F()
849 EXPECT_EQ("[4]", get_attr_kernel->shape_list[1].DebugString()); in TEST_F()
/external/tensorflow/tensorflow/python/eager/
Dbackprop_test.py971 shape_list = [[], None, [1, 2, 3], [None, None], [1, None, 3]]
973 [tensor_shape.TensorShape(s).as_proto() for s in shape_list],
974 backprop.make_attr([pywrap_tensorflow.TF_ATTR_SHAPE], shape_list))
/external/tensorflow/tensorflow/python/ops/
Dimage_ops_impl.py1770 shape_list = ([array_ops.ones(rank_1, dtype=dtypes.int32)] +
1772 multiples = array_ops.concat(shape_list, 0)