OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:component_shape
(Results
1 - 8
of
8
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/python_api/
xla_shape.py
61
for
component_shape
in self._tuple_shapes:
63
component_message.CopyFrom(
component_shape
.message)
/external/tensorflow/tensorflow/core/kernels/data/experimental/
map_and_batch_dataset_op.cc
533
TensorShape
component_shape
({dataset()->batch_size_});
534
component_shape
.AppendShape(return_values->at(i).shape());
539
component_shape
);
578
TensorShape
component_shape
(result->output[i].shape());
579
component_shape
.set_dim(0, result->num_elements);
583
component_shape
);
691
TensorShape
component_shape
(t.shape());
692
component_shape
.set_dim(0, dataset()->batch_size_);
695
Tensor new_t(ctx->allocator(attr), t.dtype(),
component_shape
);
[
all
...]
numa_map_and_batch_dataset_op.cc
565
TensorShape
component_shape
(
567
component_shape
.set_dim(0, batches_[next_output_].error_index);
572
batches_[next_output_].outputs[i].dtype(),
component_shape
);
[
all
...]
/external/tensorflow/tensorflow/core/kernels/data/
padded_batch_dataset_op.cc
329
TensorShape
component_shape
({});
331
component_shape
.AddDim(batch_component_shape.dim_size(i));
335
if (batch_elements[i][component_index].shape() ==
component_shape
) {
/cts/apps/CameraITS/tests/scene4/
test_aspect_ratio_and_crop.py
542
child_shape =
component_shape
(ct)
544
prt_shape =
component_shape
(contours[parent])
645
def
component_shape
(contour):
function
test_multi_camera_alignment.py
135
child_shape =
component_shape
(ct)
137
prt_shape =
component_shape
(contours[parent])
182
def
component_shape
(contour):
function
/external/tensorflow/tensorflow/core/kernels/
barrier_ops.cc
162
TensorShape
component_shape
(ready_tuples[0][i].shape());
163
component_shape
.InsertDim(0, insertion_size);
164
Tensor component(ready_tuples[0][i].dtype(),
component_shape
);
/external/tensorflow/tensorflow/compiler/xla/service/
elemental_ir_emitter.cc
[
all
...]
Completed in 146 milliseconds