/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
H A D | 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;
|
H A D | 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(),
|
H A D | 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 [all...] |
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
H A D | 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/ |
H A D | 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));
|
H A D | 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...] |
/external/tensorflow/tensorflow/contrib/resampler/kernels/ |
H A D | 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 variable [all...] |
/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | 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, 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 D | 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 750 0, ShapeUtil::MakeTupleShape({data_shape, data_shape}), "param0")); 753 HloInstruction::CreateGetTupleElement(data_shape, tuple_param [all...] |
H A D | hlo_instruction_test.cc | 1147 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 D | range_coder_ops.cc | 47 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 D | 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/compiler/xla/tests/ |
H A D | 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});
|