HomeSort by relevance Sort by last modified time
    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