HomeSort by relevance Sort by last modified time
    Searched defs:original_shape (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/tests/
replay_test.cc 54 std::unique_ptr<ProgramShape> original_shape = local
58 ASSERT_TRUE(protobuf_util::ProtobufEquals(*original_shape, *replayed_shape));
86 std::unique_ptr<ProgramShape> original_shape = local
90 ASSERT_TRUE(protobuf_util::ProtobufEquals(*original_shape, *replayed_shape));
132 std::unique_ptr<ProgramShape> original_shape = local
136 ASSERT_TRUE(protobuf_util::ProtobufEquals(*original_shape, *replayed_shape));
  /external/tensorflow/tensorflow/compiler/xla/service/
bfloat16_normalization.cc 194 auto original_shape = crs->shape(); local
219 *tuple->mutable_shape() = original_shape;
  /external/tensorflow/tensorflow/core/kernels/
strided_slice_op.cc 318 TensorShape original_shape = old_lhs.shape(); variable
  /external/tensorflow/tensorflow/compiler/xla/client/
computation_builder.cc 433 std::unique_ptr<Shape> original_shape = shape_or_status.ConsumeValueOrDie(); local
435 VLOG(3) << "original shape: " << ShapeUtil::HumanString(*original_shape);
446 for (int i = 0; i < ShapeUtil::Rank(*original_shape); ++i) {
448 new_sizes.push_back(original_shape->dimensions(i));
450 new_sizes.back() *= original_shape->dimensions(i);
    [all...]

Completed in 1083 milliseconds