Home
last modified time | relevance | path

Searched defs:output_shape_vec (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_to_dense_op.cc88 auto output_shape_vec = output_shape.flat<Index>(); in Compute() local
Dreshape_util.cc164 auto output_shape_vec = result_shape->vec<int64>(); in ReshapeSparseTensor() local
Ddeserialize_sparse_variant_op.cc111 auto output_shape_vec = output_shape->vec<int64>(); in Compute() local
/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_mat_mul_op.cc155 auto output_shape_vec = output_shape.vec<int64>(); in Compute() local
/external/tensorflow/tensorflow/core/kernels/image/
Dscale_and_translate_op.cc293 auto output_shape_vec = output_shape_t.vec<int32>(); in Compute() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc2219 std::vector<Shape> output_shape_vec; in InferReduceWindowShape() local