Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Ddata_format.cc62 std::vector<int64> expanded_shape = in Expand() local
/external/tensorflow/tensorflow/core/kernels/
Ddeserialize_sparse_string_op.cc128 Tensor expanded_shape(DT_INT64, TensorShape({1 + rank})); in Compute() local
Dsparse_tensors_map_ops.cc429 Tensor expanded_shape(DT_INT64, TensorShape({1 + rank})); in Compute() local
/external/tensorflow/tensorflow/lite/tools/optimize/sparsity/
Dformat_converter.cc87 std::vector<int> expanded_shape(num_expanded_dims); in DenseToSparse() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_group_converter.cc107 Shape expanded_shape = shape; in ExpandedFilterShape() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc1207 SmallVector<Attribute, 4> expanded_shape( in ExpandedShape() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc2779 auto expanded_shape = input_hlos[i]->shape(); in HandleReduce() local