| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| shape_util.cc | 30 int64 dim_size = input_shape.dim_size(i); local
31 if (!FastBoundsCheck(dim_size, std::numeric_limits<int32>::max())) {
34 " but dim ", i, " is ", dim_size);
36 vec(i) = static_cast<int32>(dim_size);
41 int64 dim_size = input_shape.dim_size(i); local
42 vec(i) = dim_size;
|
| mirror_pad_op.cc | 45 int64 dim_size = original_shape.dimensions(dimno); local
50 lhs_padding <= dim_size - excluded_edges);
52 rhs_padding <= dim_size - excluded_edges);
55 xla::SliceInDim(t_rev, dim_size - excluded_edges - lhs_padding,
56 dim_size - excluded_edges, 1, dimno);
78 TensorShapeUtils::IsMatrix(pad_shape) && pad_shape.dim_size(1) == 2,
82 ctx, dims == pad_shape.dim_size(0),
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| shape_partition.cc | 22 // Gather outer-most dims where dim_size >= 'target_partition_count'.
116 const int64 dim_size = shape_.dimensions(dimensions_[i]); local
118 std::max(int64{1}, dim_size / dimension_partition_counts_[i]);
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| resolve_constant_slice.cc | 54 int dim_size = size[i]; local
55 if (dim_size == -1) {
57 dim_size = input_array.shape().dims()[i] - begin[i];
59 CHECK_GE(dim_size, 1);
60 end.push_back(begin[i] + dim_size - 1);
|
| propagate_fixed_sizes.cc | 1686 int dim_size = local
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| shape_inference_testutil.cc | 249 int64 dim_size = 0; local
252 !strings::safe_strto64(match, &dim_size)) {
256 dims.push_back(manager->MakeDim(dim_size));
|
| tensor_shape.cc | 59 if (kIsPartial && proto.unknown_rank()) return proto.dim_size() == 0;
80 if (proto.dim_size() > 0) {
272 int64 TensorShapeBase<Shape>::dim_size(int d) const { function in class:tensorflow::TensorShapeBase
460 if (dim_size(d) != b.dim_size(d)) return false;
472 proto->add_dim()->set_size(dim_size(i));
516 int64 dim = shape.dim_size(i);
531 if (proto.dim_size() == 0) return s;
552 if (shape.dim_size(i) != prefix.dim_size(i)) return false
[all...] |
| tensor.h | 210 int64 dim_size(int d) const { return shape().dim_size(d); } function in class:tensorflow::Tensor
281 /// REQUIRES: `0 <= dim0_start <= dim0_limit <= dim_size(0)`
297 /// REQUIRES: `0 <= dim0_start < dim_size(0)`
[all...] |
| tensor_shape_test.cc | 48 EXPECT_EQ(20, s.dim_size(0));
53 EXPECT_EQ(2, s.dim_size(1));
82 EXPECT_EQ(2, s.dim_size(0));
100 EXPECT_EQ(7, s0.dim_size(0));
113 ASSERT_EQ(7, s2.dim_size(0));
166 EXPECT_EQ(10, s.dim_size(0));
167 EXPECT_EQ(2147483648, s.dim_size(1));
171 EXPECT_EQ(10, s2.dim_size(0));
172 EXPECT_EQ(2147483648, s2.dim_size(1));
269 int64 dim_size(int d) const function in class:tensorflow::__anon45008::TensorShapeOld
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| shape_ops.h | 52 int64 dim_size = shape.dim_size(i); variable
55 ctx, FastBoundsCheck(dim_size, std::numeric_limits<int32>::max()),
57 i, " is ", dim_size));
59 vec(i) = static_cast<OutType>(dim_size);
81 int64 dim_size = shape.dim_size(j); variable
84 ctx, FastBoundsCheck(dim_size, std::numeric_limits<int32>::max()),
86 i, " dim ", j, " is ", dim_size));
88 vec(j) = static_cast<OutType>(dim_size);
[all...] |
| transpose_op.cc | 135 // 2) output.dim_size(i) == input.dim_size(perm[i]);
172 const auto dim_size = input.dim_size(d); local
173 shape.AddDim(dim_size);
|
| set_kernels.cc | 105 const auto dim_size = sparse_tensor_shape[j]; local
107 ctx, dim_size > 0,
108 errors::Internal("Invalid dim_size[", j, "] = ", dim_size, "."));
111 OP_REQUIRES(ctx, dim_size > index,
113 dim_size, ", got ", index, "."));
185 out_shape_flat(i) = output_shape.dim_size(i);
216 const auto end = start + input_shape.dim_size(input_shape.dims() - 1);
418 for (int i = 0; i < t.dims(); ++i) vec[i] = t.dim_size(i);
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| grpc_tensor_coding.cc | 99 int64 dim_size = val.shape().dim_size(d); local
103 core::VarintLength(dim_size);
111 int64 dim_size = val.shape().dim_size(d); local
113 core::VarintLength(dim_size);
115 e->WriteUint64(TensorShapeProto_Dim::kSizeFieldNumber, dim_size);
|
| /external/tensorflow/tensorflow/core/util/ |
| tensor_format.h | 515 auto dim_size = spatial[dim]; local
518 CHECK_EQ(0, dim_size % 4)
520 << dim_size;
522 dim_size /= 4;
524 dim_sizes[GetTensorSpatialDimIndex(dims, format, dim)] = dim_size;
|
| mkl_util.h | 243 int64 dim_size(int index) const { return sizes_[index]; } function in class:tensorflow::MklShape
[all...] |
| /external/tensorflow/tensorflow/python/eager/ |
| pywrap_tensor.cc | 1077 tensorflow::int64 dim_size = TFE_TensorDebugInfoOnDeviceDim(debug_info, i); local
[all...] |
