Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Ddeserialize_sparse_variant_op.cc111 auto output_shape_vec = output_shape->vec<int64>(); in Compute() local
112 output_shape_vec(0) = num_sparse_tensors; in Compute()
114 output_shape_vec(j) = input.dim_size(j); in Compute()
117 output_shape_vec(j + input_dims_to_stack) = element_shape_vec(j); in Compute()
146 output_shape_vec(j + input_dims_to_stack) = std::max( in Compute()
147 output_shape_vec(j + input_dims_to_stack), element_shape_vec(j)); in Compute()
Dshape_ops.h168 absl::InlinedVector<int64, 8> output_shape_vec(input_dims + 1); in Compute()
170 output_shape_vec[i] = input_shape.dim_size(i); in Compute()
172 output_shape_vec[dim] = 1; in Compute()
174 output_shape_vec[i] = input_shape.dim_size(i - 1); in Compute()
176 TensorShape output_shape(output_shape_vec); in Compute()
Dsparse_to_dense_op.cc88 auto output_shape_vec = output_shape.flat<Index>(); in Compute() local
90 OP_REQUIRES_OK(c, TensorShapeUtils::MakeShape(output_shape_vec.data(), in Compute()
91 output_shape_vec.size(), in Compute()
Dreshape_util.cc164 auto output_shape_vec = result_shape->vec<int64>(); in ReshapeSparseTensor() local
166 output_shape_vec(j) = output_shape.dim_size(j); in ReshapeSparseTensor()
/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_mat_mul_op.cc155 auto output_shape_vec = output_shape.vec<int64>(); in Compute() local
156 if (rank == 3) output_shape_vec(0) = batch_size; in Compute()
157 output_shape_vec(row_dim) = a_shape.dim_size(row_dim); in Compute()
158 output_shape_vec(row_dim + 1) = b_shape.dim_size(row_dim + 1); in Compute()
174 const int64 num_output_rows = output_shape_vec(row_dim); in Compute()
/external/tensorflow/tensorflow/core/kernels/image/
Dscale_and_translate_op.cc293 auto output_shape_vec = output_shape_t.vec<int32>(); in Compute() local
294 const int64 output_height = internal::SubtleMustCopy(output_shape_vec(0)); in Compute()
295 const int64 output_width = internal::SubtleMustCopy(output_shape_vec(1)); in Compute()
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc2219 std::vector<Shape> output_shape_vec; in InferReduceWindowShape() local
2224 output_shape_vec.push_back(cur_output_shape); in InferReduceWindowShape()
2227 CHECK_EQ(output_shape_vec.size(), 1); in InferReduceWindowShape()
2228 return output_shape_vec[0]; in InferReduceWindowShape()
2230 return ShapeUtil::MakeTupleShape(output_shape_vec); in InferReduceWindowShape()