OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:tuple_shapes_size
(Results
1 - 16
of
16
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/service/
tuple_util.cc
49
tuple_elements.reserve(input_shape.
tuple_shapes_size
());
50
for (int i = 0; i < input_shape.
tuple_shapes_size
(); i++) {
while_util.cc
51
narrow_shape.
tuple_shapes_size
());
75
wide_parameter, narrow_shape.
tuple_shapes_size
());
81
for (int i = narrow_shape.
tuple_shapes_size
();
82
i < wide_shape.
tuple_shapes_size
(); i++) {
102
int64 elements_in_old_while_shape = while_instr->shape().
tuple_shapes_size
();
129
new_while, while_instr->shape().
tuple_shapes_size
())));
135
i < new_while_shape.
tuple_shapes_size
(); i++) {
190
for (int64 i = 1, e = loop_state_shape.
tuple_shapes_size
(); i < e; i++) {
while_loop_simplifier.cc
329
for (int64 i = 0; i < while_shape.
tuple_shapes_size
(); ++i) {
346
for (int64 i = 0; i < while_shape.
tuple_shapes_size
(); ++i) {
367
for (int64 i = 0; i < while_shape.
tuple_shapes_size
(); ++i) {
382
for (int64 i = 0; i < while_shape.
tuple_shapes_size
(); ++i) {
592
for (int64 i = 0; i < desired_shape.
tuple_shapes_size
(); ++i) {
630
for (int64 i = 0; i < shape.
tuple_shapes_size
(); ++i) {
684
for (int64 i = 0; i < flat_shape.
tuple_shapes_size
(); ++i) {
[
all
...]
dynamic_dimension_inference.cc
467
int64 operand_count = hlo->shape().
tuple_shapes_size
();
dynamic_dimension_inference_test.cc
531
EXPECT_EQ(while_hlo->shape().
tuple_shapes_size
(), 3);
shape_inference.cc
[
all
...]
/external/tensorflow/tensorflow/compiler/xla/
shape.cc
49
tuple_shapes_.reserve(shape_proto.
tuple_shapes_size
());
68
proto.mutable_tuple_shapes()->Reserve(
tuple_shapes_size
());
shape.h
112
int
tuple_shapes_size
() const { return tuple_shapes_.size(); }
function in class:xla::Shape
shape_util.cc
340
return shape.
tuple_shapes_size
();
596
return pointer_size * shape.
tuple_shapes_size
();
640
if (shape.
tuple_shapes_size
() > 0) {
773
if (!subshape->IsTuple() || i >= subshape->
tuple_shapes_size
() || i < 0) {
797
i >= return_shape->
tuple_shapes_size
()) {
[
all
...]
literal_comparison.cc
[
all
...]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_batchnorm_thunk.cc
155
CHECK_EQ(hlo->shape().
tuple_shapes_size
(), 3);
237
CHECK_EQ(hlo->shape().
tuple_shapes_size
(), 3);
/external/tensorflow/tensorflow/compiler/aot/
codegen.cc
215
size_t num_results = ps.result().
tuple_shapes_size
();
220
ps.result().
tuple_shapes_size
(), ")");
267
size_t num_results = ps.result().
tuple_shapes_size
();
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_transfer_manager.cc
206
for (int64 i = 0; i < literal_shape.
tuple_shapes_size
(); ++i) {
ir_emitter.cc
354
for (int64 i = 0; i < data_shape.
tuple_shapes_size
(); ++i) {
491
for (int64 i = 0; i < operand_shape.
tuple_shapes_size
(); ++i) {
[
all
...]
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
llvm_util.cc
227
result_type = llvm::ArrayType::get(result_type, shape.
tuple_shapes_size
());
/external/tensorflow/tensorflow/compiler/xla/client/
xla_builder.cc
334
if (shape->
tuple_shapes_size
() != 0) {
335
for (int64 i = 0; i < shape->
tuple_shapes_size
(); ++i) {
[
all
...]
Completed in 119 milliseconds