Searched defs:data_shape (Results 1 - 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dsegment_reduction_ops.cc44 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;
H A Ddynamic_stitch_op.cc55 // 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(),
H A Dreduction_ops_common.cc50 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
[all...]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dinstruction_fusion_test.cc142 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/
H A Ddata_flow_ops.cc58 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));
H A Dmath_ops.cc860 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...]
/external/tensorflow/tensorflow/contrib/resampler/kernels/
H A Dresampler_ops.cc135 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 variable
[all...]
/external/tensorflow/tensorflow/compiler/xla/service/
H A Dliveness_util_test.cc85 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, tupl
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...]
H A Dbuffer_liveness_test.cc617 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
750 0, ShapeUtil::MakeTupleShape({data_shape, data_shape}), "param0"));
753 HloInstruction::CreateGetTupleElement(data_shape, tuple_param
[all...]
H A Dhlo_instruction_test.cc1147 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); local
1153 HloInstruction::CreateTranspose(data_shape, b, {1, 0}));
1158 HloInstruction::CreateDot(data_shape, a, b_t, dot_dnums));
1162 HloInstruction::CreateBroadcast(data_shape, one, {1}));
1164 data_shape, HloOpcode::kAdd, dot, add_operand));
1166 data_shape, HloOpcode::kSubtract, dot, add_operand));
1168 HloInstruction::CreateBinary(data_shape, HloOpcode::kMultiply, add, sub));
/external/tensorflow/tensorflow/contrib/coder/kernels/
H A Drange_coder_ops.cc47 BroadcastRange(T* data_pointer, gtl::ArraySlice<int64> data_shape, argument
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, argument
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,
182 RangeEncodeImpl(TTypes<int16>::ConstFlat data, gtl::ArraySlice<int64> data_shape, TTypes<int32>::ConstMatrix cdf, gtl::ArraySlice<int64> cdf_shape, string* output) const argument
190 BroadcastRange<const int16, int32, N> view{data.data(), data_shape, local
242 std::vector<int64> data_shape, cdf_shape; variable
[all...]
H A Drange_coder_ops_test.cc136 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/compiler/xla/tests/
H A Dtuple_test.cc433 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});

Completed in 719 milliseconds