Searched refs:dynamic_dim_size (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | fill_op.cc | 59 auto dynamic_dim_size = xla::Slice(ctx->Input(0), {i}, {i + 1}, {1}); in Compile() local
60 dynamic_dim_size = xla::Reshape(dynamic_dim_size, {}); in Compile()
61 dynamic_dim_size = xla::ConvertElementType(dynamic_dim_size, xla::S32); in Compile()
62 output = xla::SetDimensionSize(output, dynamic_dim_size, i); in Compile()
|
| D | tile_ops.cc | 94 auto dynamic_dim_size = in Compile() local
96 dynamic_dim_size = xla::Reshape(dynamic_dim_size, {}); in Compile()
97 dynamic_dim_size = xla::ConvertElementType(dynamic_dim_size, xla::S32); in Compile()
98 result = xla::SetDimensionSize(result, dynamic_dim_size, i); in Compile()
|
| D | tensor_list_ops.cc | 74 auto dynamic_dim_size = xla::Slice(ctx->Input(0), {dim}, {dim + 1}, {1}); in GetTensorListDynamicDims() local
75 dynamic_dim_size = xla::Reshape(dynamic_dim_size, {}); in GetTensorListDynamicDims()
76 dynamic_dim_size = xla::ConvertElementType(dynamic_dim_size, xla::S32); in GetTensorListDynamicDims()
77 dynamic_dims.push_back(dynamic_dim_size); in GetTensorListDynamicDims()
|