Searched refs:dims_are_dynamic (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reshape_op.cc | 120 std::vector<bool> dims_are_dynamic; in Compile() local
126 ctx, ctx->ResolveInputDynamismIntoPredVector(1, &dims_are_dynamic)); in Compile()
131 shape.dim_sizes(), dims_are_dynamic)); in Compile()
168 dims_are_dynamic[unknown_index] = input_is_dynamic; in Compile()
173 shape.dim_sizes(), dims_are_dynamic)); in Compile()
176 << ", dynamic_dims=" << xla::VectorString(dims_are_dynamic); in Compile()
|
| D | tensor_list_ops.cc | 57 std::vector<bool> dims_are_dynamic; in GetTensorListDynamicDims() local
59 ctx->ResolveInputDynamismIntoPredVector(0, &dims_are_dynamic)); in GetTensorListDynamicDims()
73 if (dims_are_dynamic[dim]) { in GetTensorListDynamicDims()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 260 const std::vector<bool>& dims_are_dynamic);
|
| D | shape_inference.cc | 2914 const std::vector<bool>& dims_are_dynamic) { in InferDynamicReshapeShape() argument
2915 if (new_size_bounds.size() != dims_are_dynamic.size()) { in InferDynamicReshapeShape()
2919 new_size_bounds.size(), dims_are_dynamic.size()); in InferDynamicReshapeShape()
2931 operand.element_type(), new_size_bounds, dims_are_dynamic); in InferDynamicReshapeShape()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 480 const std::vector<bool>& dims_are_dynamic);
1082 const std::vector<bool>& dims_are_dynamic);
1696 const std::vector<bool>& dims_are_dynamic);
|
| D | xla_builder.cc | 1150 const std::vector<bool>& dims_are_dynamic) { in DynamicReshape() argument
1162 new_size_bounds, dims_are_dynamic)); in DynamicReshape()
4139 const std::vector<bool>& dims_are_dynamic) { in DynamicReshape() argument
4141 dims_are_dynamic); in DynamicReshape()
|