/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | replay_test.cc | 54 std::unique_ptr<ProgramShape> original_shape = in TEST_F() local 58 ASSERT_TRUE(protobuf_util::ProtobufEquals(original_shape->ToProto(), in TEST_F() 87 std::unique_ptr<ProgramShape> original_shape = in XLA_TEST_F() local 91 ASSERT_TRUE(protobuf_util::ProtobufEquals(original_shape->ToProto(), in XLA_TEST_F() 134 std::unique_ptr<ProgramShape> original_shape = in TEST_F() local 138 ASSERT_TRUE(protobuf_util::ProtobufEquals(original_shape->ToProto(), in TEST_F()
|
/external/tensorflow/tensorflow/contrib/periodic_resample/kernels/ |
D | periodic_resample_op.h | 269 const tensorflow::TensorShape& original_shape, in do_periodic_resample_op() argument 286 process_desired_shape(context, original_shape, desired_shape.dim_sizes(), in do_periodic_resample_op() 302 output_shape = original_shape; in do_periodic_resample_op() 315 InputIndexer input_indexer(target_dimensions, original_shape, in do_periodic_resample_op() 368 const tensorflow::TensorShape& original_shape, in create_grad_tensor() argument 373 context, original_shape, desired_shape, grad_tensor); \ in create_grad_tensor() 408 context->GetAttr("original_shape", &original_shape)); in PeriodicResampleOpGrad() 415 create_grad_tensor(context, grad_tensor, grad_tensor_type, original_shape, in Compute() 420 tensorflow::TensorShape original_shape;
|
/external/tensorflow/tensorflow/contrib/periodic_resample/ops/ |
D | array_ops.cc | 147 tensorflow::TensorShape original_shape; in __anon96b295990202() local 148 TF_RETURN_IF_ERROR(c->GetAttr("original_shape", &original_shape)); in __anon96b295990202() 150 TF_RETURN_IF_ERROR(c->MakeShapeFromTensorShape(original_shape, &s)); in __anon96b295990202()
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | mirror_pad_op.cc | 30 const xla::Shape& original_shape, in DoMirrorPad() argument 41 for (int64 dimno = original_shape.rank() - 1; dimno >= 0; --dimno) { in DoMirrorPad() 45 int64 dim_size = original_shape.dimensions(dimno); in DoMirrorPad()
|
/external/tensorflow/tensorflow/python/data/experimental/ops/ |
D | scan_ops.py | 109 for original_shape, weakened_shape in zip(flat_state_shapes, 111 if original_shape.ndims is not None and ( 113 original_shape.as_list() != weakened_shape.as_list()):
|
D | grouping.py | 334 for original_shape, weakened_shape in zip(flat_state_shapes, 336 if original_shape.ndims is not None and ( 338 original_shape.as_list() != weakened_shape.as_list()):
|
D | batching.py | 539 for original_shape, new_shape in zip(flat_original_shapes, 541 if not original_shape.is_compatible_with(new_shape):
|
/external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/ |
D | reshape_vectorizer.cc | 34 Output GetVectorizedShape(Scope* s, Output tensor, Output original_shape) { in GetVectorizedShape() argument 44 return ops::Concat(*s, {dim_0, original_shape}, ops::Const(*s, 0)); in GetVectorizedShape()
|
/external/tensorflow/tensorflow/contrib/data/python/ops/ |
D | batching.py | 204 original_shape.merge_with(new_shape) 205 for original_shape, new_shape in zip(flat_original_shapes,
|
/external/tensorflow/tensorflow/python/ops/ |
D | embedding_ops.py | 762 original_shape = sparse_ids.dense_shape 766 array_ops.size(original_shape) 771 array_ops.slice(original_shape, [0], [original_rank - 1])), 772 array_ops.gather(original_shape, original_rank - 1)]) 817 math_ops.cast(original_shape, dtypes.int32), [0],
|
/external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
D | batch_reshape.py | 373 def calculate_reshape(original_shape, new_shape, validate=False, name=None): argument 378 with ops.name_scope(name, "calculate_reshape", [original_shape, new_shape]): 379 original_size = math_ops.reduce_prod(original_shape) 388 original_shape, 1, message="Original shape must be a vector."),
|
/external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/ |
D | resnet_preprocessing.py | 96 original_shape = tf.image.extract_jpeg_shape(image_bytes) 97 bad = _at_least_x_are_equal(original_shape, tf.shape(image), 3)
|
/external/tensorflow/tensorflow/python/ops/ragged/ |
D | ragged_tensor_shape_test.py | 303 original_shape = RaggedTensorDynamicShape.from_dim_sizes(original_dim_sizes) 305 self.assertEqual(original_shape.rank, bcast_shape.rank) 307 bcast1 = original_shape.broadcast_dimension(axis, row_length)
|
/external/tensorflow/tensorflow/contrib/layers/python/layers/ |
D | embedding_ops.py | 126 original_shape = sparse_ids.dense_shape 130 array_ops.size(original_shape) 135 array_ops.slice(original_shape, [0], [original_rank - 1])), 136 array_ops.gather(original_shape, original_rank - 1)]) 181 math_ops.cast(original_shape, dtypes.int32), [0],
|
D | layers.py | 268 original_shape = inputs.get_shape() 270 original_rank = original_shape.ndims 422 if original_shape.ndims == 2:
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | bfloat16_normalization.cc | 237 auto original_shape = hlo->shape(); in HandleMultipleOutputs() local 265 *tuple->mutable_shape() = original_shape; in HandleMultipleOutputs()
|
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | op_level_cost_estimator.h | 29 TensorShapeProto MaybeGetMinimumShape(const TensorShapeProto& original_shape,
|
D | op_level_cost_estimator.cc | 183 TensorShapeProto MaybeGetMinimumShape(const TensorShapeProto& original_shape, in MaybeGetMinimumShape() argument 185 auto shape = original_shape; in MaybeGetMinimumShape() 203 shape.add_dim()->set_size(original_shape.dim(i).size()); in MaybeGetMinimumShape()
|
/external/tensorflow/tensorflow/python/keras/layers/ |
D | normalization.py | 576 original_shape = [-1] + inputs.shape.as_list()[1:] 577 expanded_shape = [self.virtual_batch_size, -1] + original_shape[1:] 583 outputs = array_ops.reshape(outputs, original_shape)
|
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | array_ops_test.py | 946 original_shape = constant_op.constant([6, 4, 4], dtype=dtypes.int64) 951 dx = array_ops.strided_slice_grad(original_shape, begin, end, strides, 961 original_shape = constant_op.constant([6, 4, 4], dtype=dtypes.int64) 969 dx = array_ops.strided_slice_grad(original_shape, begin, end, strides,
|
/external/tensorflow/tensorflow/python/keras/ |
D | backend.py | 2409 original_shape = int_shape(x) 2426 if original_shape[rows] is None: 2429 new_height = original_shape[rows] * height_factor 2431 if original_shape[cols] is None: 2434 new_width = original_shape[cols] * width_factor
|
/external/tensorflow/tensorflow/core/kernels/ |
D | strided_slice_op.cc | 333 TensorShape original_shape = old_lhs->shape(); in Compute() local
|
/external/tensorflow/tensorflow/python/data/ops/ |
D | dataset_ops.py | 1323 for original_shape, weakened_shape in zip(flat_state_shapes, 1325 if original_shape.ndims is not None and ( 1327 original_shape.as_list() != weakened_shape.as_list()):
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder.cc | 935 TF_ASSIGN_OR_RETURN(const Shape& original_shape, GetShape(operand)); in Collapse() 937 VLOG(3) << "original shape: " << ShapeUtil::HumanString(original_shape); in Collapse() 941 for (int i = 0; i < original_shape.rank(); ++i) { in Collapse() 943 new_sizes.push_back(original_shape.dimensions(i)); in Collapse() 945 new_sizes.back() *= original_shape.dimensions(i); in Collapse()
|