| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| segment_reduction_ops.cc | 44 TensorShape data_shape = ctx->InputShape(0); variable
52 OP_REQUIRES(ctx, data_shape.dims() >= indices_shape.dims(),
58 OP_REQUIRES(ctx, (data_shape.dim_size(d) == indices_shape.dim_size(d)),
61 " of data_shape, but dimension ",
62 d, " differs ", data_shape.dim_size(d), " vs. ",
66 TensorShape buffer_shape = data_shape;
|
| reduction_ops_common.cc | 50 const TensorShape data_shape = ctx->InputShape(0); local
68 VLOG(1) << "data shape: " << data_shape.DebugString();
71 gtl::InlinedVector<bool, 4> bitmap(data_shape.dims(), false);
77 !(index < -data_shape.dims() || index >= data_shape.dims()),
79 " for input with ", data_shape.dims(),
81 index = (index + data_shape.dims()) % data_shape.dims();
84 num_elements_reduced *= data_shape.dim_size(index);
88 for (int i = 0; i < data_shape.dims(); ++i)
[all...] |
| dynamic_stitch_op.cc | 55 // Validate that data_shape[i] = indices[i].shape() + constant
74 const TensorShape& data_shape = data_shapes[input_num]; variable
76 ctx, TensorShapeUtils::StartsWith(data_shape, indices_shape),
78 "].shape = ", data_shape.DebugString(),
84 data_shape, indices_shape),
89 input_num, "].shape = ", data_shape.DebugString(),
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| liveness_util_test.cc | 85 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); local
87 0, ShapeUtil::MakeTupleShape({data_shape, data_shape}), "tuple"));
89 HloInstruction::CreateGetTupleElement(data_shape, tuple, 0));
91 HloInstruction::CreateGetTupleElement(data_shape, tuple, 1));
100 data_shape, gte1, update, starts));
199 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); local
201 0, ShapeUtil::MakeTupleShape({data_shape, data_shape}), "tuple"));
203 HloInstruction::CreateGetTupleElement(data_shape, tuple, 0))
239 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); local
272 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); local
308 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); local
350 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); local
380 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); local
[all...] |
| buffer_liveness_test.cc | 617 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); local
620 0, ShapeUtil::MakeTupleShape({data_shape, data_shape}), "param0"));
623 HloInstruction::CreateGetTupleElement(data_shape, tuple_param0, 0));
626 HloInstruction::CreateGetTupleElement(data_shape, tuple_param0, 1));
643 data_shape, gte1, update, starts));
747 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); local
[all...] |
| hlo_instruction_test.cc | 1147 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); local
[all...] |
| /external/tensorflow/tensorflow/contrib/coder/kernels/ |
| range_coder_ops.cc | 47 BroadcastRange(T* data_pointer, gtl::ArraySlice<int64> data_shape,
50 CHECK(!data_shape.empty());
51 CHECK_EQ(data_shape.size(), N);
54 std::copy(data_shape.begin(), data_shape.end(), &data_shape_[0]);
118 Status CheckCdfShape(const TensorShape& data_shape,
120 if (TF_PREDICT_FALSE(cdf_shape.dims() != data_shape.dims() + 1)) {
123 data_shape.DebugString(), ", cdf shape=", cdf_shape.DebugString());
150 std::vector<int64> data_shape, cdf_shape; variable
152 context, MergeAxes(data.shape(), cdf.shape(), &data_shape, &cdf_shape))
190 BroadcastRange<const int16, int32, N> view{data.data(), data_shape, local
242 std::vector<int64> data_shape, cdf_shape; variable
[all...] |
| range_coder_ops_test.cc | 136 const TensorShape& data_shape = data.shape(); local
137 Tensor shape{DT_INT32, {data_shape.dims()}};
138 for (int i = 0; i < data_shape.dims(); ++i) {
139 shape.flat<int32>()(i) = data_shape.dim_size(i);
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| data_flow_grad_test.cc | 50 TensorShape data_shape({2, 3, 2});
51 auto data = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(data_shape));
55 RunTest({data}, {data_shape}, y.outputs,
|
| /external/tensorflow/tensorflow/contrib/resampler/python/ops/ |
| resampler_ops_test.py | 137 data_shape = (batch_size, data_height, data_width, data_channels)
138 data = np.random.rand(*data_shape).astype(dtype.as_numpy_dtype)
169 data_shape = (batch_size, data_height, data_width, data_channels)
170 data = np.random.rand(*data_shape).astype(dtype.as_numpy_dtype)
214 data_shape = (batch_size, data_height, data_width, data_depth,
216 data = np.zeros(data_shape)
229 data_shape = (batch_size, data_height, data_width, data_channels)
230 data = np.zeros(data_shape)
242 data_shape = (batch_size, data_height, data_width, data_channels)
243 data = np.zeros(data_shape)
[all...] |
| /external/tensorflow/tensorflow/contrib/resampler/kernels/ |
| resampler_ops.cc | 135 const ::tensorflow::TensorShape& data_shape = data.shape(); variable
136 OP_REQUIRES(ctx, data_shape.dims() == 4,
141 data_shape.DebugString()));
154 OP_REQUIRES(ctx, data_shape.dim_size(0) == warp_shape.dim_size(0),
158 data_shape.DebugString(), ", ", warp_shape.DebugString()));
159 const int batch_size = data_shape.dim_size(0);
160 const int data_height = data_shape.dim_size(1);
161 const int data_width = data_shape.dim_size(2);
162 const int data_channels = data_shape.dim_size(3);
340 const ::tensorflow::TensorShape& data_shape = data.shape() variable
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| instruction_fusion_test.cc | 142 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); local
143 Shape tuple_shape = ShapeUtil::MakeTupleShape({data_shape, data_shape});
147 HloInstruction::CreateGetTupleElement(data_shape, param, 0));
149 HloInstruction::CreateGetTupleElement(data_shape, param, 1));
151 HloInstruction::CreateBinary(data_shape, HloOpcode::kAdd, gte0, gte1));
|
| /external/tensorflow/tensorflow/core/ops/ |
| math_ops.cc | 860 ShapeHandle data_shape; local
862 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &data_shape));
866 TF_RETURN_IF_ERROR(c->Subshape(data_shape, 1, &subshape));
876 ShapeHandle data_shape; local
877 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &data_shape));
890 TF_RETURN_IF_ERROR(c->Subshape(data_shape, 1, &subshape));
900 ShapeHandle data_shape; local
901 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &data_shape));
914 TF_RETURN_IF_ERROR(c->Subshape(data_shape, 1, &subshape));
938 ShapeHandle data_shape; local
[all...] |
| data_flow_ops.cc | 58 ShapeHandle data_shape = c->input(0);
70 c->MergePrefix(data_shape, partitions_shape, &unused, &unused));
73 // data_shape in the remaining dimensions.
77 TF_RETURN_IF_ERROR(c->Subshape(data_shape, rank, &data_suffix_shape));
105 ShapeHandle data_shape = c->input(i + num_partitions); local
111 // Assert that data_shape starts with indices_shape.
114 c->MergePrefix(data_shape, indices_shape, &unused, &unused));
118 TF_RETURN_IF_ERROR(c->Subshape(data_shape, indices_rank, &rest));
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| tuple_test.cc | 433 Shape data_shape = ShapeUtil::MakeShape(F32, {3}); local
434 Shape inner_tuple_shape = ShapeUtil::MakeTupleShape({data_shape, data_shape});
436 ShapeUtil::MakeTupleShape({inner_tuple_shape, data_shape});
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| xla.proto | 437 Shape data_shape = 2;
|
| /external/tensorflow/tensorflow/python/keras/_impl/keras/engine/ |
| training.py | 139 data_shape = data[i].shape
145 'with shape ' + str(data_shape))
147 data_shape = data_shape[1:]
149 for dim, ref_dim in zip(data_shape, shape):
154 ' but got array with shape ' + str(data_shape))
[all...] |
