Home
last modified time | relevance | path

Searched defs:expanded_shape (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Ddata_format.cc60 std::vector<int64> expanded_shape(input_shape.dimensions()); in Expand() local
/external/tensorflow/tensorflow/core/kernels/
Ddeserialize_sparse_string_op.cc126 Tensor expanded_shape(DT_INT64, TensorShape({1 + rank})); in Compute() local
Dsparse_tensors_map_ops.cc428 Tensor expanded_shape(DT_INT64, TensorShape({1 + rank})); in Compute() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.cc49 xla::Shape expanded_shape = shape; in ExpandedFilterShapeForDepthwiseConvolution() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_group_converter.cc108 Shape expanded_shape = shape; in ExpandedFilterShape() local